import torch
# data
TRAIN_BATCH_SIZE = 256
TEST_BATCH_SIZE = 256

# model
IMG_SIZE=224
IMG_CHANNEL=3
PATCH_SIZE=16
EMB_DIM=256
HEAD_NUM=8
MLP_RATIO=4
DEPTH=12
CLASS_NUM=10
HEAD_DIM=64

# train
#BATCH_SIZE = 64
DEVICE = torch.device("cuda" if torch.cuda.is_available() else 'cpu')
EPOCHS = 114
LEARNING_RATE = 0.01
LEARNING_RATE_F = 0.001
MOMENTUM = 0.9
GAMMA = 0.9
STEP_SIZE = 10
SAVE_STEP = EPOCHS